square of number in python